from selenium import webdriver
from pathlib import Path
from time import sleep, time

# 生成年月日时分秒时间
picture_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
directory_time = time.strftime("%Y-%m-%d", time.localtime(time.time()))
print(picture_time)
print(directory_time)



driver = webdriver.Chrome()
driver.get("http://www.baidu.com")
driver.find_element_by_id('kw').send_keys('selenium')
driver.find_element_by_id('su').click()
sleep(5)
# 截取当前窗口，并指定截图图片的保存位置

i=1
scrpath="C:\\Users\\\zhouk\\Desktop\\测试数据\\截图文件"  #指定的保存目录
capturename = '\\'+str(i) + '.png'  #自定义命名截图
wholepath=scrpath+capturename
print (wholepath)
if Path(scrpath).is_dir():  #判断文件夹路径是否已经存在
    pass
else:
    Path(scrpath).mkdir()   #如果不存在，创建文件夹
while Path(wholepath).exists():   #判断文件是否已经存在，也可使用is_file()判断
    i+=1
    capturename = '\\'+str(i) + '.png'
    wholepath = scrpath+capturename
driver.get_screenshot_as_file(wholepath) #不能接受Path类的值，只能是字符串，否则无法截图
sleep(5)
driver.quit()